SageMaker Studio スペースにタグを付与する

aws sagemaker add-tags \
--resource-arn resource-arn-to-attach-tags \
--tags Key=key, Value=value  

aws sagemaker describe-space --space-name <スペース名> --domain-id <ドメインID>

{
  "DomainId": "XXX",
  "SpaceArn": "arn:aws:sagemaker:ap-northeast-1:<アカウントID>:space/<ドメインID>/sampleA",
  "SpaceName": "sampleA",
  "Status": "InService",
  "LastModifiedTime": "2024-11-07T22:20:34.087000+00:00",
  "CreationTime": "2024-11-07T22:20:28.818000+00:00",
  "SpaceSettings": {
    "AppType": "JupyterLab",
    "SpaceStorageSettings": {
      "EbsStorageSettings": {
        "EbsVolumeSizeInGb": 5
      }
    }
  },
  "OwnershipSettings": {
    "OwnerUserProfileName": "default-20241108T071632"
  },
  "SpaceSharingSettings": {
    "SharingType": "Private"
  },
  "Url": "https://xxx.studio.ap-northeast-1.sagemaker.aws/jupyterlab/default"
}

aws sagemaker add-tags --resource-arn <1. で取得したARN> --tags Key=CmBillingGroup,Value=<設定したい値>

{
  "Tags": [
    {
      "Key": "CmBillingGroup",
      "Value": "space-sampleA"
    }
  ]
}

JupyterLabSpace:
    Type: AWS::SageMaker::Space
    Properties:
      DomainId: !GetAtt StudioDomain.DomainId
      SpaceName: !Ref SpaceName
      SpaceDisplayName: JupyterLab
      OwnershipSettings:
        OwnerUserProfileName: !Ref UserProfileName
      SpaceSharingSettings:
        SharingType: Private
      SpaceSettings:
        AppType: JupyterLab
        JupyterLabAppSettings:
          DefaultResourceSpec:
            InstanceType: !Ref InstanceType
            SageMakerImageVersionArn: XXX
        SpaceStorageSettings:
          EbsStorageSettings:
            EbsVolumeSizeInGb: !Ref EbsVolumeSizeGb
      # タグを設定可能
      Tags:
        - Key: CmBillingGroup
          Value: space-sampleA

#!/bin/bash

# 変数定義
## ドメインIDを設定（要設定）
DOMAIN_ID="xxx"
## Space名を設定（要設定）
SPACE_NAME="YYY"
EXPECTED_TAG_KEY="CmBillingGroup"
EXPECTED_TAG_VALUE="space-${SPACE_NAME}"

# Space の情報を取得
echo "Checking space details..."
SPACE_ARN=$(aws sagemaker describe-space \
    --space-name "$SPACE_NAME" \
    --domain-id "$DOMAIN_ID" \
    --query 'SpaceArn' \
    --output text)

if [ -z "$SPACE_ARN" ]; then
    echo "Error: Could not get Space ARN"
    exit 1
fi

# タグを追加
echo "Adding tag..."
aws sagemaker add-tags \
    --resource-arn "$SPACE_ARN" \
    --tags "Key=$EXPECTED_TAG_KEY,Value=$EXPECTED_TAG_VALUE"

# タグが正しく設定されたか確認
echo "Verifying tags..."
TAG_VALUE=$(aws sagemaker list-tags \
    --resource-arn "$SPACE_ARN" \
    --query "Tags[?Key=='$EXPECTED_TAG_KEY'].Value" \
    --output text)

if [ "$TAG_VALUE" == "$EXPECTED_TAG_VALUE" ]; then
    echo "Completed: Tag verification successful"
    echo "Tag Key: $EXPECTED_TAG_KEY"
    echo "Tag Value: $TAG_VALUE"
    exit 0
else
    echo "Error: Tag verification failed"
    echo "Expected: $EXPECTED_TAG_VALUE"
    echo "Got: $TAG_VALUE"
    exit 1
fi